作者投稿和查稿 主编审稿 专家审稿 编委审稿 远程编辑

计算机工程

• •    

基于扩散模型的生成对抗式推荐方法

  • 出版日期:2026-04-08 发布日期:2026-04-08

Generative Adversarial Recommendation Method Based on Diffusion Model

  • Online:2026-04-08 Published:2026-04-08

摘要: 生成式模型因其有效的数据生成能力,近年来被广泛应用于推荐系统领域。然而,现有生成式推荐系统由于模型生成过程的随机性导致稳定性不足,并且有限的表示学习能力影响了个性化推荐的准确性。为了解决上述问题,提出一种基于扩散模型的生成对抗式推荐方法。具体来说,首先通过变分自编码器 (VAE) 对原始向量进行压缩,然后使用扩散模型在隐式空间进行多步加噪和去噪,学习高质量用户表示。此外,引入对抗训练机制为去噪过程提出反馈信号,缓解其生成过程不可控的问题。在Amazon-book、Yelp和Movielens-1M三个公开数据集上进行实验,所提出的方法相比主要基线在召回率 (Recall@10)和归一化折损累计增益 (NDCG@10)上分别最高提升20.3%、18.9%,说明方法能够有效预测用户行为,提高推荐精度。

Abstract: Generative models have achieved remarkable results due to their effective data generation capabilities and have been widely applied in the field of recommendation systems in recent years. Generative recommendation, through probabilistic modeling, directly learns the potential distribution of users' historical behaviors and generates possible interaction scenarios, breaking through the traditional retrieval paradigm and becoming a research hotspot in the field of recommendation systems. However, the existing generative recommendation systems have insufficient stability due to the randomness of the model generation process, and the limited representation learning ability affects the accuracy of personalized recommendations. To solve the above problems, a generative adversarial recommendation method based on diffusion model is proposed. Specifically, first of all, to alleviate the resource consumption caused by direct diffusion, the original vector is compressed through a Variational Autoencoder (VAE). Then, the diffusion model is used in latent space for multi-step denoising noising and denoising to learn high-quality user representations. In addition, an adversarial training mechanism is introduced to provide feedback signals for the denoising process, alleviating the problem of uncontrollability in its generation process. Experiments were conducted on three public datasets, namely Amazon-book, Yelp and Movielens-1M. Compared with the baseline model, the proposed method has significant improvements in the metrics of Recall and normalized cumulative loss gain (NDCG), indicating that the method can effectively predict user behavior. Improve the accuracy of recommendations.